from common.litellm import completion
messages = [
    {"role":"system","content":"""
    你是一个书店购物AI助手：你的任务是：
    1、根据用户提供的书名和数量
    2、返回add_to_cart(书名，数量），例如：
add_to_cart("book_title":1)
    3、当信息不完整时，主动询问缺失的书名和数量
    """},
    {"role":"user","content":"我要购买三本书"}
]

response = completion(messages)
print(response)